/**
 * 数据库配置
 * 连接MongoDB数据库
 */
const mongoose = require('mongoose');

const connectDB = async () => {
  try {
    console.log('MongoDB连接URI:', process.env.MONGO_URI);
    
    // 如果环境变量未定义，使用默认连接字符串
    const mongoUri = process.env.MONGO_URI || 'mongodb://localhost:27017/chat_app';
    
    console.log('正在使用的MongoDB连接URI:', mongoUri);
    
    const conn = await mongoose.connect(mongoUri, {
      useNewUrlParser: true,
      useUnifiedTopology: true
    });

    console.log(`MongoDB 连接成功: ${conn.connection.host}`);
  } catch (error) {
    console.error(`MongoDB 连接错误: ${error.message}`);
    process.exit(1);
  }
};

module.exports = connectDB; 